package leetcode.editor.cn;

//给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。 
//
// 
//
// 示例 1： 
//
// 
//
// 
//输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
//输出：6
//解释：上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。 
// 
//
// 示例 2： 
//
// 
//输入：height = [4,2,0,3,2,5]
//输出：9
// 
//
// 
//
// 提示： 
//
// 
// n == height.length 
// 1 <= n <= 2 * 10⁴ 
// 0 <= height[i] <= 10⁵ 
// 
// Related Topics 栈 数组 双指针 动态规划 单调栈 👍 3379 👎 0

import java.util.Arrays;
import java.util.Stack;

//Java：【42】 - 接雨水
public class TrappingRainWater_42{
    public static void main(String[] args) {
        Solution solution = new TrappingRainWater_42().new Solution();
                // TO TEST
    }
    
    //leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    public int trap(int[] height) {
        if(height.length<3) return 0;
        int sum=0;
        Stack<Integer> left = new Stack<>();
        int current = 0;
        while (current<height.length){
            while (!left.isEmpty() && height[current]>height[left.peek()]){
                int top = left.pop();
                if(left.isEmpty()) break;
                int distance = current - left.peek() -1;
                int water_height = Math.min(height[current], height[left.peek()]) - height[top];
                sum+=distance*water_height;
            }
            left.push(current++);
        }
        return sum;
    }
}
//leetcode submit region end(Prohibit modification and deletion)

    // 动态编程
public int trap(int[] height) {
    if(height.length<3) return 0;
    int sum=0;
    int[] left_max = new int[height.length];
    int[] right_max = new int[height.length];
    left_max[0] = height[0]; // 注意初始化，两边未必为0
    for (int i = 1; i < height.length; i++) {
        left_max[i] = Math.max(left_max[i-1],height[i]);
    }
    right_max[height.length-1] = height[height.length-1];// 注意初始化，两边未必为0
    for (int i = height.length-2; i >=0 ; i--) {
        right_max[i] = Math.max(right_max[i+1], height[i]);
    }
//        System.out.println(Arrays.toString(height));
//        System.out.println(Arrays.toString(left_max));
//        System.out.println(Arrays.toString(right_max));
    for (int i = 0; i < height.length; i++) {
        sum+=Math.min(left_max[i], right_max[i])-height[i];
    }
    return sum;
}
}
